package cn.tedu.weibo.controller;

import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.util.UUID;

@RestController
public class UploadController {

    @RequestMapping("/upload")
    public String upload(MultipartFile pic) throws IOException {
        //pic 代表客户端上传的文件参数
        System.out.println("pic = " + pic);

        //得到图片的原始文件名
        String fileName=pic.getOriginalFilename();
        System.out.println("原始文件名"+fileName);

        //得到原始文件名的后缀
        String suffix=fileName.substring(fileName.lastIndexOf("."));
        //得到唯一文件名UUID.randomUUID()获取唯一标识符
        fileName=UUID.randomUUID()+suffix;
        System.out.println(fileName);
        //准备保存图片的路径
        String dirPath="d:/files";
        File dirFile=new File(dirPath);
        //判断若果不存在,将文件夹创建出来
        if(!dirFile.exists()){
            dirFile.mkdirs();//创建文件夹
        }
        //得到文件的完整路径
        String filePath=dirPath+"/"+fileName;
        //把图片保存到上面的路径中 异常抛出
        pic.transferTo(new File(filePath));

        //把图片的url路径响应给客户端 /文件名
        return "/"+fileName;
    }
    @RequestMapping("/remove")
    public void remove(String url){
        new File("d:/files"+url).delete();
    }

}
